package com.sise.Tree;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

/**
 *      剑指 Offer 32 - I. 从上到下打印二叉树
 *
 *      从上到下打印出二叉树的每个节点，同一层的节点按照从左到右的顺序打印。
 *      给定二叉树:[3,9,20,null,null,15,7],
 *
 *         3
 *        / \
 *       9  20
 *         /  \
 *        15   7
 *
 *      返回：[3,9,20,15,7]
 */
public class _offer_32_1_levelOrder {
    public int[] levelOrder(TreeNode root) {
        if (root == null) return new int[0];

        List<Integer> list = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);

        while (!queue.isEmpty()){
            int size = queue.size();
//            for (int i = 0; i < size; i++){       // 在结果集需要分组的情况下，就需要得到当前层次的数值个数，但在此题中并无要求 [[3],[20,9],[15,7]]，故此可以取消
                TreeNode poll = queue.poll();
                list.add(poll.val);
                if (poll.left != null){
                    queue.add(poll.left);
                }
                if (poll.right != null){
                    queue.add(poll.right);
                }
//            }
        }
        int[] res = new int[list.size()];
        for (int i = 0; i < list.size(); i++){
            res[i] = list.get(i);
        }
        return res;
    }
}
